from flask_records.decorators import query_by_page, query
from common.utils.utils import custom_marshal_pagination

GET_BY_PAGE_SQL = """
select *
from unit
where (is_active = :is_active or :is_active is null)
  and (
            lower(unit.name) like lower(:search_str_re) or
            lower(unit.notes) like lower(:search_str_re) or
            :search_str_re is null
    )   
    and (unit.id not in :filter_ids or :filter_ids is null)    
"""


def get_by_page(page, page_size, sorted_by, sorted_by_type, search_str, **kwargs):
    sql = GET_BY_PAGE_SQL
    order_by_sql = "{} ORDER BY {} {}".format(sql, sorted_by, sorted_by_type)

    if search_str:
        # 以空格分割，搜索所有包含的
        search_str_re = '%{}%'.format(search_str)
    else:
        search_str_re = None

    filter_ids = kwargs.pop('filter_ids', None)
    if filter_ids is None:
        filter_ids = tuple([None])
    else:
        filter_ids = tuple(filter_ids)

    @query_by_page(order_by_sql, page_size=page_size)
    def _get_by_pagination(
            page, search_str_re, filter_ids=None, is_active=None):
        pass

    pagination = _get_by_pagination(
        page, search_str_re, filter_ids, **kwargs
    )
    pagination = custom_marshal_pagination(pagination)
    return pagination
